package com.cy.server.config;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.cy.server.pojo.MailConstants;
import com.cy.server.pojo.MailLog;
import com.cy.server.service.IMailLogService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @peoject: yeb
 * @description: rabbitMQ配置类
 * @author: LiShaoZhuo
 * @create: 2021-05-08 11:05
 **/
@Configuration
@Slf4j
public class RabbitMQConfig {

    @Autowired
    CachingConnectionFactory connectionFactory;
    @Autowired
    IMailLogService mailLogService;

    @Bean
    public RabbitTemplate rabbitTemplate(){
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        
        /** 
        * @Description: 消息确认回调，确认消息是否到达broker
        * @Param:
         * data: 消息唯一标识
         * ack: 确认结果
         * cause: 失败原因
        * @return: org.springframework.amqp.rabbit.core.RabbitTemplate 
        * @Author: LiShaoZhuo 
        * @Date: 2021/5/8 
        */
        rabbitTemplate.setConfirmCallback((data,ack,cause)->{
            String msgId = data.getId();
            if (ack){
                log.info("{}==============>消息发送成功",msgId);
                mailLogService.update(new UpdateWrapper<MailLog>().set("status", 1).eq("msgId", msgId));
            }else {
                log.error("{}==============>消息发送失败",msgId);
            }
        });
        
        /** 
        * @Description: 消息失败回调
        * @Param:
         * msg: 消息主题
         * repCode: 响应码
         * repText: 响应描述
         * exchange: 交换机
         * routingKey: 路由键
        * @return: org.springframework.amqp.rabbit.core.RabbitTemplate 
        * @Author: LiShaoZhuo 
        * @Date: 2021/5/8 
        */
        rabbitTemplate.setReturnCallback((msg,repCode,repText,exchange,routingKey)->{
            log.error("{}==============>消息发送queue时失败",msg.getBody());
        });
        return rabbitTemplate;
    }

    @Bean
    public Queue queue(){
        return new Queue(MailConstants.MAIL_QUEUE_NAME);
    }

    @Bean
    public DirectExchange directExchange(){
        return new DirectExchange(MailConstants.MAIL_EXCHANGE_NAME);
    }

    @Bean
    public Binding binding(){
        return BindingBuilder.bind(queue()).to(directExchange()).with(MailConstants.MAIL_ROUTING_KEY_NAME);
    }
}
